package com.gaogzhen.algorithms4.sort;

import edu.princeton.cs.algs4.*;

/**
 * 比较2个算法用时
 */
public class SortCompare {

    /**
     * 计算某个算法用时
     * @param alg       算法名称
     * @param a         随机数组
     * @return          时间
     */
    public static double time(String alg, Double[] a) {
        Stopwatch timer = new Stopwatch();
        if (alg.equals("Insertion")) Insertion.sort(a);
        if (alg.equals("Selection")) Selection.sort(a);
        if (alg.equals("Shell")) Shell.sort(a);
        if (alg.equals("Merge")) Merge.sort(a);
        if (alg.equals("Quick")) Quick.sort(a);
        if (alg.equals("Heap")) Heap.sort(a);
        return timer.elapsedTime();
    }

    /**
     * 测试T次长度为N的数组排序用时
     * @param alg      算法名称
     * @param N        数组长度
     * @param T        测试次数
     * @return          总用时
     */
    public static double timeRandomInput(String alg, int N, int T) {
        // 使用算法alg将T个长度为N的数组排序
        double total = 0.0;
        Double[] a = new Double[N];
        for (int t = 0; t < T; t++) {
            //进行一次测试（生成一个数组并排序
            for (int i = 0; i < N; i++)
                a[i] = StdRandom.uniform();
            total += time(alg, a);
        }

        return total;
    }

    public static void main(String[] args) {
        String alg1 = args[0];
        String alg2 = args[1];

        int N = Integer.parseInt(args[2]);
        int T = Integer.parseInt(args[3]);

        // 算法1总时间
        double t1 = timeRandomInput(alg1, N, T);
        // 算法2总时间
        double t2 = timeRandomInput(alg2, N, T);

        StdOut.printf("For %d random Doubles\n %s is ", N, alg1);
        StdOut.printf("%.1f times faster then %s\n", t2/t1, alg2);
    }
}
